#ATM机的存取款功能
#ATM角色
#用户功能

#1.先登录
#根据用户登录

#2.菜单界面
    # 1.取款
    #     用户账户金额就减少了500
    # 2.存款
    #     用户的账户金额增加了500
    # 3.查询
    #     用户的账户金额显示


def check_user(username,passwd):  #判断用户名和密码
    user = {"root":"123456","llf":"123456"}
    tuple1 = (username,passwd)
    if tuple1 in user.items():
        return True
    else:
        return False

def menu():
    ATM_sum = 10000   #ATM里有10000块金额
    user_account = {"root":5000,"llf":5000}
    while True:
        print("取款请输入'1',存款请输入'2',查询请输入'3',退出请输q")
        choice = input("请输入您的选项：")
        if choice == "1":
            money = float(input("请输入您要取款的金额："))
            if user_account[username] >= money:
                user_account[username] -= money
                ATM_sum -= money
                print("取款成功！")
            else:
                print("您的余额不足，取款失败")
        elif choice == "2":
            money = float(input("请输入您要存款的金额："))
            user_account[username] += money
            ATM_sum += money
            print("存款成功！")
        elif choice == "3":
            print(f"{username}的余额{user_account[username]}元")
        else:
            print("已退出菜单界面")
            break

while True:
    print("退出请输入q")
    username = input("请输入用户名：")
    passwd = input("请输入密码：")
    if check_user(username,passwd):
        print(f"欢迎{username}登录")
        break
    elif username == 'q' or passwd == 'q':
        print("已退出登录系统")
        break
    else:
        print("用户名密码输入错误，请重新输入")

if username != 'q' and passwd != 'q':
    menu()